from csagent.core.node.base_node import BaseNode
from csagent.core.context import AgentState
import time
class Node(BaseNode):
    def initialize(self):
        """初始化节点"""
        # 如果节点存在初始化操作，需要实现该函数，否则忽略
        # self.conf为节点配置中的conf文件内容
        # 如配置为空则为{}
        print(self.conf) 
        print('hello Node初始化成功')
    def execute(self, state: AgentState, stream_writer=None) -> AgentState:
        # stream_writer功能是需要流式传输结果的时候，比如大模型问答，详见相关demo
        print('hello world')
        state['messages'].append({
            "role":"assistant",
            "content":"hello world"
        })
        state['extra']["用户任意定义的key"] = 'hello world'
        return state